Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,6 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
|
32 |
|
33 |
# Load text-only model and tokenizer (Pocket Llama)
|
34 |
model_id = "prithivMLmods/Pocket-Llama2-3.2-3B-Instruct"
|
35 |
-
#model_id = "prithivMLmods/Flerovium-2B"
|
36 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
37 |
model = AutoModelForCausalLM.from_pretrained(
|
38 |
model_id,
|
@@ -286,12 +285,7 @@ demo = gr.ChatInterface(
|
|
286 |
["Write the code that converts temperatures between Celsius and Fahrenheit in short"],
|
287 |
[{"text": "Create a short story based on the image.", "files": ["examples/1.jpg"]}],
|
288 |
["@JennyNeural Who was Nikola Tesla and what were his contributions?"],
|
289 |
-
[{"text": "@video-infer Describe the video", "files": ["examples/Missing.mp4"]}]
|
290 |
-
# [{"text": "@video-infer Describe the Ad", "files": ["examples/coca.mp4"]}],
|
291 |
-
# ["@GuyNeural Explain how rainbows are formed."],
|
292 |
-
# ["@PalomaNeural What is the water cycle?"],
|
293 |
-
# ["@AlonsoNeural Who was Pablo Picasso and why is he famous?"],
|
294 |
-
# ["@MadhurNeural What are the key principles of Ayurveda?"]
|
295 |
],
|
296 |
cache_examples=False,
|
297 |
description="# **Callisto OCR**",
|
|
|
32 |
|
33 |
# Load text-only model and tokenizer (Pocket Llama)
|
34 |
model_id = "prithivMLmods/Pocket-Llama2-3.2-3B-Instruct"
|
|
|
35 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
36 |
model = AutoModelForCausalLM.from_pretrained(
|
37 |
model_id,
|
|
|
285 |
["Write the code that converts temperatures between Celsius and Fahrenheit in short"],
|
286 |
[{"text": "Create a short story based on the image.", "files": ["examples/1.jpg"]}],
|
287 |
["@JennyNeural Who was Nikola Tesla and what were his contributions?"],
|
288 |
+
[{"text": "@video-infer Describe the video", "files": ["examples/Missing.mp4"]}]
|
|
|
|
|
|
|
|
|
|
|
289 |
],
|
290 |
cache_examples=False,
|
291 |
description="# **Callisto OCR**",
|