K00B404 commited on
Commit
e467758
·
verified ·
1 Parent(s): 5a5e741

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -0
app.py ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Use a pipeline as a high-level helper
2
+ from transformers import pipeline
3
+ premium_models = [
4
+ "HuggingFaceH4/zephyr-7b-beta",
5
+ "K00B404/BagOClownCoders-slerp-7B",
6
+ "Qwen/Qwen2.5-Omni-7B",
7
+ "Qwen/Qwen2.5-VL-7B-Instruct",
8
+ "deepseek-ai/Janus-Pro-7B",
9
+ "meta-llama/Llama-2-7b-hf",
10
+ "Alibaba-NLP/gte-Qwen2-7B-instruct",
11
+
12
+ ]
13
+ messages = [
14
+ {"role": "system_prompt", "content": "You are a ChatBuddy and chat with the user in a Human way"},
15
+ {"role": "user", "content": "Who are you?"}
16
+ ]
17
+ pipe = pipeline("text-generation", model=premium_models[0])
18
+ pipe(messages)