Spaces:

msaifee
/

LLaMA-3-2-Text-Generator

Paused

msaifee commited on Sep 26

Commit

20ea8ef

•

1 Parent(s): bf86fa8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,6 +2,8 @@ import os
 import gradio as gr
 from huggingface_hub import login
 from transformers import AutoModelForCausalLM, AutoTokenizer
 # Fetch API token from environment variable
 api_token = os.getenv("Llama_Token")

 import gradio as gr
 from huggingface_hub import login
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
+from transformers import pipeline
 # Fetch API token from environment variable
 api_token = os.getenv("Llama_Token")