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**",
|